package com.atguigu.cloud.mygateway;

import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * 自定义全局过滤器
 * （继承接口：
 * GlobalFilter:这是一个Spring Cloud Gateway中的全局过滤器接口。
 *  Ordered:提供了排序的能力，用于控制过滤器的执行顺序。
 *  ）
 *
 * 统计接口调用时间
 *
 * @author 23087
 */
@Component
@Slf4j
public class MyGlobalFilter implements GlobalFilter, Ordered
{
    // 开始调用方法的时间
    public static final String BEGIN_VISIT_TIME = "begin_visit_time";

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // 1. 先记录下访问接口的开始时间
        exchange.getAttributes().put(BEGIN_VISIT_TIME,System.currentTimeMillis());

        // 2. 返回统计的各个接口给后台
        return chain.filter(exchange).then(Mono.fromRunnable(() -> {
            Long beginVisitTime = exchange.getAttribute(BEGIN_VISIT_TIME);
            if (beginVisitTime !=null){
                log.info("访问接口主机：{}", exchange.getRequest().getURI().getHost());
                log.info("访问接口端口：{}", exchange.getRequest().getURI().getPort());
                log.info("访问接口URL：{}", exchange.getRequest().getURI().getPath());
                log.info("访问接口URL后面的参数：{}", exchange.getRequest().getURI().getRawQuery());
                log.info("访问接口时长：{}{}", System.currentTimeMillis() - beginVisitTime ,"ms");
                System.out.println("================ 分割线 =========================");

            }
        }));

    }

    /**
     * 数字越小，优先级越高
     * @return
     */
    @Override
    public int getOrder() {
        return 0;
    }
}
